// 实现异步日志缓冲区

#ifndef __M_BUF_H__
#define __M_BUF_H__

#include <vector>
#include <cassert>

namespace superlog
{
    #define DEFAULT_BUFFER_SIZE (1*1024*1024)   //默认大小1M
    #define THRESHOLD_BUFFER_SIZE (8*1024*1024) //设置阈值为8M
    #define INCREMENT_BUFFER_SIZE (1*1024*1024) //线性增长1M

    class Buffer
    {
    public:
        Buffer()
            : _buffer(DEFAULT_BUFFER_SIZE), //缓冲区默认大小
              _writer_idx(0),
              _reader_idx(0)
        {}

        // 向缓冲区中写入数据
        void push(const char* data, size_t len)
        {
            // 缓冲区剩余空间不够的情况
            // 1. 阻塞/返回false -- 无法添加(主要用于实际使用，因为资源不能无限制使用)
            //if(len > writeAbleSize()) return; //固定大小，则直接返回

            // 上面这种情况无需再考虑，因为在外面(AsyncLooper工作器)已经通过条件判断了
            // 2. 扩容(动态空间，主要用于测试，一般是测试极限性能，空间不足就扩容)

            ensureEnoughSize(len);
            
            // 1.将数据拷贝进缓冲区
            std::copy(data, data+len, &_buffer[_writer_idx]);
            // 2.将当前写入位置向后偏移
            moveWriter(len);
        }

        // 返回当前可写空间的大小
        size_t writeAbleSize()
        {
            // 对于当前的扩容思路来说，不存在可写空间大小，因为总是可写，因此这个接口仅仅针对固定大小缓冲区提供
            return (_buffer.size() - _writer_idx);
        }

        // 返回可读数据起始地址
        const char* begin()
        {
            return &_buffer[_reader_idx];
        }

        // 返回可读数据的长度
        size_t readAbleSize()
        {
            // 因为当前实现的缓冲区并非循环缓冲区，而是双缓冲区，处理完就进行交换，所以不存在空间循环使用
            return (_writer_idx - _reader_idx);
        }

        // 对读指针进行向后偏移操作
        void moveReader(size_t len)
        {
            assert(len <= readAbleSize());
            _reader_idx += len;
        }

        // 重置读写位置，初始化缓冲区
        void reset()
        {
            _writer_idx = 0; //缓冲区所有的空间都是空闲的
            _reader_idx = 0; //与_writer_idx相等，表示没有数据可读
        }

        // 对Buffer实现交换操作
        void swap(Buffer& buffer)
        {
            _buffer.swap(buffer._buffer);
            std::swap(_reader_idx, buffer._reader_idx);
            std::swap(_writer_idx, buffer._writer_idx);
        }
        
        // 判断缓冲区是否为空
        bool empty()
        {
            return (_reader_idx == _writer_idx);
        }
    private:
        // 对空间进行扩容
        void ensureEnoughSize(size_t len)
        {
            if(len <= writeAbleSize()) return; //不需要扩容
            size_t new_size = 0;
            // 完善：此处也可修改为用循环(判断new_size-_buffer.size()是否大于等于len，不满足就继续循环)
            // 这里目前只做简化操作(直接+len)
            if(_buffer.size() < THRESHOLD_BUFFER_SIZE)
            {
                new_size = _buffer.size()*2 + len; //小于阈值则翻倍增长(+len是因为即使*2也有可能还是小于len)
            }
            else
            {
                new_size = _buffer.size() + INCREMENT_BUFFER_SIZE + len; //否则线性增长(+len是因为即使加上线性增长值也有可能还是小于len)
            }
            _buffer.resize(new_size);
        }
        // 对写指针进行向后偏移操作
        void moveWriter(size_t len) //向缓冲区中写入数据时，直接内部调用即可
        {
            assert(len + _writer_idx <= _buffer.size());
            _writer_idx += len;
        }
    private:
        std::vector<char> _buffer; //缓冲区
        size_t _writer_idx; //当前可写数据的指针(本质是下标)
        size_t _reader_idx; //当前可读数据的指针(本质是下标)
    };
}

#endif